package cn.iocoder.devlop.module.annual.report.handler;

import cn.iocoder.devlop.module.annual.report.utils.ReportYearHelper;
import com.baomidou.mybatisplus.extension.plugins.handler.TableNameHandler;

import java.util.ArrayList;
import java.util.List;

/**
 * @author Euan
 * @since 2024/1/22
 */
public class TableNameSuffixHandler implements TableNameHandler {

    @Override
    public String dynamicTableName(String sql, String tableName) {
        List<String> list = new ArrayList<>();
        list.add("t_baseinfo");
        list.add("t_ws_baseinfo");
        list.add("t_cu_entreginfo");
        list.add("t_subcapital");
        list.add("t_forinvestment");
        list.add("t_forguaranteeinfo");
        list.add("t_websiteinfo");
        list.add("t_alterstockinfo");
        list.add("t_overduepay");
        list.add("t_sfc_branchinfo");
        list.add("t_sfc_licenceinfo");
        list.add("t_ws_subcapital");
        list.add("t_socsecinfo");
        list.add("t_sup_baseinfo");
        if (list.contains(tableName)) {
            return tableName + "_" + ReportYearHelper.getStartYear();
        }
        return tableName;
    }
}
